from utils import create_input_files

if __name__ == '__main__':
    # Create input files (along with word map)
    # create_input_files(dataset='coco',
    #                    karpathy_json_path='/data1/zyx/yks/dataset/coco/dataset_coco.json',
    #                    image_folder='/data1/zyx/yks/dataset/coco/images',
    #                    captions_per_image=5,
    #                    min_word_freq=5,
    #                    output_folder='/data1/zyx/yks/dataset/coco/caption data',
    #                    max_len=50)

    # create_input_files(dataset='coco',
    #                    karpathy_json_path="/data1/zyx/yks/object_detection/gluon-rfcn/lib/data/format.json",
    #                    image_folder='',
    #                    captions_per_image=1,
    #                    min_word_freq=5,
    #                    output_folder='/data1/zyx/yks/dataset/images_and_formula/caption',
    #                    max_len=50)
    create_input_files(dataset='coco',
                       karpathy_json_path="data_im2markup/annotations.json",
                       image_folder='',
                       captions_per_image=1,
                       min_word_freq=5,
                       output_folder='./data_im2markup',
                       max_len=64)